"""
简单表名称替换，通过 excel 配置完成
"""
import logging
import os
import pandas as pd

excel_file_name = os.getcwd() + "/mysql_pg_table_replace_excel.xlsx"


def rep_table_name_by_excel(document_content):
    """
    :Description: 通过在 excel 中配置的表名批量替换文本中内容
    :param document_content: 需要修改的字符串
    :return: 替换后的 文档内容 document_content
    """
    logging.info("通过在 excel 中配置的表名 批量 将mysql的表名称 替换为 PG 中的表名称 开始 ")
    df = pd.read_excel(excel_file_name)
    for x in df.itertuples():
        mysql_table_str = x.mysql_table
        pg_table_str = x.pg_table
        if mysql_table_str and pg_table_str:
            document_content = document_content.replace(mysql_table_str, pg_table_str)
    logging.info("通过在 excel 中配置的表名 批量 将mysql的表名称 替换为 PG 中的表名称 结束")
    return document_content
